# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/22 09:21 
@Author  : ZhangShenao 
@File    : 3.删除数据.py
@Desc    : 删除Pinecone中的向量数据


"""
import os

import dotenv
from langchain_openai import OpenAIEmbeddings
from langchain_pinecone import PineconeVectorStore

# 加载环境变量
dotenv.load_dotenv()

# 创建Embeddings
embeddings = OpenAIEmbeddings(
    model='text-embedding-3-small',  # 选择text-embedding-3-small模型
    openai_api_base=os.getenv('OPENAI_API_BASE')
)

# 创建Pinecone向量数据库
vector_store = PineconeVectorStore(
    index_name='llm-ops',  # 指定索引名称
    embedding=embeddings,  # 指定Embedding模型
    namespace='llm-ops',  # 指定Namespace
    text_key='text',  # 指定文本属性key
)

# 根据ID删除向量
vector_store.delete(ids=['d2806f44-d94f-4eec-9d11-06921a2c18db'])
